package com.tanqin.service;

import com.alibaba.dubbo.config.annotation.Reference;
import com.tanqin.pojo.Permission;
import com.tanqin.pojo.Role;
import com.tanqin.pojo.User;
import com.tanqin.service.UserService;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;


@Component
public class SpringSecurityUserService implements UserDetailsService {
    @Reference
    private UserService userService;

    // 根据用户名查询用户信息
    public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
        // 查询用户信息
        User user = userService.findByUsername(username);
        if (user == null) {
            // 用户名不存在
            return null;
        } else {
            // 给用户授权
            List<GrantedAuthority> list = new ArrayList<>();
            Set<Role> roles = user.getRoles();
            if (roles != null && roles.size() > 0) {
                for (Role role : roles) {
                    // 授予角色
                    list.add(new SimpleGrantedAuthority(role.getKeyword()));
                    Set<Permission> permissions = role.getPermissions();
                    if (permissions != null && permissions.size() > 0) {
                        for (Permission permission : role.getPermissions()) {
                            // 授予权限
                            list.add(new SimpleGrantedAuthority(permission.getKeyword()));
                        }
                    }
                }
            }
            // 创建 security 的 User 对象
            org.springframework.security.core.userdetails.User securityUser = new org.springframework.security.core.userdetails.User(username, user.getPassword(), list);
            return securityUser;
        }
    }
}
